package cn.zhouchao.tran;

import org.apache.rocketmq.client.producer.LocalTransactionState;
import org.apache.rocketmq.client.producer.TransactionListener;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageExt;

public class TranListener implements TransactionListener {
    /**
     * message：给消费者使用的
     * arg：执行本地事务所需要的参数
     * @param message
     * @param arg
     * @return
     */
    @Override
    public LocalTransactionState executeLocalTransaction(Message message, Object arg) {
        try {
            System.out.println("执行本地事务的方法，在此方法中进行用户新增！" + arg);
            // 此处如果本地事务执行成功，那么就返回commit
//            int i = 1/0;
            return LocalTransactionState.COMMIT_MESSAGE;
        } catch (Exception e) {
            e.printStackTrace();
            // 此处如果本地事务有异常，那么就响应rollback，告诉mq刚才的事务消息不要了
            return LocalTransactionState.ROLLBACK_MESSAGE;
        }
    }

    @Override
    public LocalTransactionState checkLocalTransaction(MessageExt messageExt) {
        System.out.println("回查本地事务");
        return LocalTransactionState.COMMIT_MESSAGE;
    }
}
